package com.itheima.controller;


import com.itheima.common.R;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.UUID;

@RestController
@RequestMapping("/common")
public class CommonController {


    @Value("${reggie.path}")
    private String bean;

    /**
     * 上传文件
     * @param file
     * @return
     */
    @PostMapping("/upload")
    public R<String> upload(MultipartFile file) {

//        获取原始文件名称
        String originalFilename = file.getOriginalFilename();
//        截取文件名后缀
        String substring = originalFilename.substring(originalFilename.indexOf("."));
//        使用UUID重新生成文件名称 并拼接后缀
        String s = UUID.randomUUID().toString()+substring;

//        创建一个文件目录
        File file1 = new File(bean);

//        判断mul是否存在
        if (!file1.exists()){
//           不存在就创建
            file1.mkdirs();
        }

//        临时文件存储到指定的位置
        try {
            file.transferTo(new File(bean + s));
        } catch (IOException e) {
            e.printStackTrace();
        }
        return R.success(s);
    }

    /**
     * 下载
     * @param name
     * @param response
     */
    @GetMapping("/download")
    public void download(String name , HttpServletResponse response){

        try {
//            输入流  读取文件内容
            FileInputStream fileInputStream = new FileInputStream(new File(bean+name));
//            输出流 将文件写回给浏览器
            ServletOutputStream outputStream = response.getOutputStream();
//            设置响应回去的是什么文件
            response.setContentType("image/jpeg");
//            读取
            int len ;
            byte[] bytes = new byte[1024];
            while ((len=fileInputStream.read(bytes)) !=-1){
//                输出
                outputStream.write(bytes,0,len);
//                刷新
                outputStream.flush();
            }
//            关流
            outputStream.close();
            fileInputStream.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }


}
